getting index of element in list (python)